export OUTPUT_DIR=/share_data/liupan/bert-finetuned


export DATASET=/share_data/liupan/med_corpus
export PRETRAINED=/share_data/liupan/bert-base-chinese

python3 mlm_train.py --epochs 128 --lr 2e-5 --wd 0.01 --bs 256 \
    --pretrained $PRETRAINED --dataset $DATASET \
    --output_dir $OUTPUT_DIR

